from sqlalchemy import create_engine
from sqlalchemy.orm import declarative_base, sessionmaker

engine = create_engine(
    "mysql+pymysql://root:mysql@localhost:3307/stock_info?charset=utf8&autocommit=false",
    max_overflow=0,  # 超过连接池大小外最多创建的连接
    pool_size=2,  # 连接池大小
    pool_timeout=30,  # 池中没有线程最多等待的时间，否则报错
    pool_recycle=-1  # 多久之后对线程池中的线程进行一次连接的回收（重置）,-1表示不重置
)
Base = declarative_base()


def get_session():
    session = sessionmaker(bind=engine)()
    return session
